package com.zy.explore.report.interfaces.assembler;

import com.zy.explore.report.infrastructure.util.DataBeanUtil;
import com.zy.explore.report.interfaces.frm.CommonIdFrm;
import com.zy.explore.report.interfaces.frm.CommonIdsFrm;
import com.zy.explore.report.interfaces.vo.PageVO;
import com.zy.explore.report.application.service.MongoExampleService;
import com.zy.explore.report.domain.mongo.entity.MongoExampleEntity;
import com.zy.explore.report.interfaces.frm.ExampleSaveFrm;
import com.zy.explore.report.interfaces.frm.ExampleQueryFrm;
import com.zy.explore.report.interfaces.vo.ExampleVO;

import org.springframework.data.domain.PageImpl;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.stereotype.Component;
import org.springframework.util.CollectionUtils;

import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;

/**
 * description Mongo示例Assembler
 *
 * @author ZY
 * @date 2021年05月15日 11:23
 */
@Component
public class MongoExampleAssembler {

    /**
     * 注入service
     */
    @Resource
    private MongoExampleService service;

    /**
     * 新增数据
     *
     * @param frm 参数对象
     * @return 视图对象
     */
    public ExampleVO insert(ExampleSaveFrm frm) {
        MongoExampleEntity entity = convertFromFrm(frm);
        entity.setCreateTime(new Date());
        entity.setUpdateTime(new Date());
        return convertToVO(service.insert(entity));
    }

    /**
     * 删除对象
     *
     * @param frm 参数对象
     * @return 1:删除成功 其他:删除失败
     */
    public Integer delete(CommonIdFrm frm) {
        return service.delete(frm.getId());
    }

    /**
     * 更新对象
     *
     * @param frm 待更新的Frm对象
     * @return 成功返回更新后的VO对象，失败返回null
     */
    public ExampleVO update(ExampleSaveFrm frm) {
        MongoExampleEntity entity = convertFromFrm(frm);
        entity.setUpdateTime(new Date());
        int res = service.update(entity);
        if (res == 1) {
            return MongoExampleAssembler.convertToVO(entity);
        }
        return null;
    }

    /**
     * 根据ID查询数据
     *
     * @param frm 主键Frm对象
     * @return 成功返回对应ID的VO对象，失败返回null
     */
    public ExampleVO findById(CommonIdFrm frm) {
        return convertToVO(service.findById(frm.getId()));
    }

    /**
     * 通过主键批量查询数据集合
     *
     * @param frm 主键集合Frm对象
     * @return 成功返回对应ID集合的VO对象集合，失败返回空集合
     */
     public List<ExampleVO> findByIds(CommonIdsFrm frm) {
        List<ExampleVO> retList = new ArrayList<>();
        if (CollectionUtils.isEmpty(frm.getIds())) {
            return retList;
        }
        List<MongoExampleEntity> list = service.findByIds(frm.getIds());
        if (!CollectionUtils.isEmpty(list)) {
            list.forEach(x -> retList.add(convertToVO(x)));
        }
        return retList;
     }

    /**
     * 根据条件查询数据集合
     *
     * @param frm 查询条件对象
     * @return 满足条件的数据集合
     */
     public List<ExampleVO> getList(ExampleQueryFrm frm) {
        List<ExampleVO> retList = new ArrayList<>();
        // 转换查询条件
        Query query = MongoExampleEntity.convertToQuery(frm);
        // 查询集合
        List<MongoExampleEntity> list = service.getList(query);
        if (!CollectionUtils.isEmpty(list)) {
            list.forEach(x -> retList.add(convertToVO(x)));
        }
        return retList;
     }

    /**
     * 根据条件查询分页
     *
     * @param frm 查询条件
     * @return 满足条件的分页
     */
    public PageVO<ExampleVO> getPageInfo(ExampleQueryFrm frm) {
        // 转换查询条件
        Query query = MongoExampleEntity.convertToQuery(frm);
        // 查询总数
        long count = service.getTotalCount(query);
        // 查询分页列表
        List<MongoExampleEntity> list = service.getPageList(query, frm.getPageNum(), frm.getPageSize());
        PageRequest pageRequest = PageRequest.of(frm.getPageNum() - 1, frm.getPageSize());
        List<ExampleVO> retList = new ArrayList<>();
        if (!CollectionUtils.isEmpty(list)) {
            list.forEach(x -> retList.add(convertToVO(x)));
        }
        return new PageVO<>(new PageImpl<>(retList, pageRequest, count));
    }

    /**
     * Frm 转 实体类
     *
     * @param frm 参数对象
     * @return 返回转换后的实体对象
     */
    public static MongoExampleEntity convertFromFrm(ExampleSaveFrm frm) {
        return DataBeanUtil.copyOf(frm, MongoExampleEntity.class, true);
    }

    /**
     * 实体类 转 VO
     *
     * @param entity 实体类对象
     * @return 返回转化后的VO对象
     */
    public static ExampleVO convertToVO(MongoExampleEntity entity) {
        return entity == null ? null : DataBeanUtil.copyOf(entity, ExampleVO.class, true);
    }

}

